package com.ss.movierecommendationsystem.base.utils;

import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * RedisTemplate 工具类
 *
 * @author wuchaocheng
 */
@Component
public class RedisTemplateUtils {

    @Resource
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 存值--带有效期--单位：毫秒
     *
     * @param key     key
     * @param data    数据
     * @param timeout 过期时间（单位：毫秒）
     */
    public void setValueTimeMS(String key, Object data, Long timeout) {
        redisTemplate.opsForValue().set(key, data, timeout, TimeUnit.MILLISECONDS);
    }

    /**
     * 存值--带有效期--单位：分钟
     *
     * @param key  key
     * @param data 数据
     * @param time 过期时间,（单位：分钟）
     */
    public void setValueTimeMIN(String key, Object data, long time) {
        redisTemplate.opsForValue().set(key, data, time, TimeUnit.MINUTES);
    }

    /**
     * 如果 key 存在则覆盖,并返回旧值.
     * 如果不存在,返回null 并添加
     *
     * @param key  key
     * @param data 数据
     * @return Object
     */
    public Object getAndSet(String key, Object data) {
        return redisTemplate.opsForValue().getAndSet(key, data);
    }

    /**
     * 存值
     *
     * @param key  key
     * @param data 数据
     */
    public void setValue(String key, Object data) {
        redisTemplate.opsForValue().set(key, data);
    }

    /**
     * 取值
     *
     * @param key key
     * @return Object
     */
    public Object getValue(String key) {
        return redisTemplate.opsForValue().get(key);
    }

    /**
     * 删除key
     *
     * @param key key
     */
    public void delKey(String key) {
        redisTemplate.delete(key);
    }

    /**
     * 删除多个key
     *
     * @param keys key
     */
    public void delKeys(List<String> keys) {
        redisTemplate.delete(keys);
    }

    /**
     * 判断key是否存在
     *
     * @param key key
     * @return Boolean
     */
    public Boolean isExist(String key) {
        return redisTemplate.hasKey(key);
    }

}
